home *** CD-ROM | disk | FTP | other *** search
/ Language/OS - Multiplatform Resource Library / LANGUAGE OS.iso / et / et-2_2.lha / et2.2 / src / IO / istream.C < prev    next >
C/C++ Source or Header  |  1990-12-06  |  6KB  |  416 lines
  1. #include "filebuf.h"
  2. #include "stream.h"
  3. #include "../CType.h"
  4. #include "../String.h"
  5.  
  6. istream cin((int)0, &cout);          // UNIX input stream 0 tied to cout
  7.  
  8. istream::istream(streambuf* s, ostream* t, bool dodelete) : stream(s, dodelete)        // bind to buffer
  9. {
  10.     tied_to= t;
  11. }
  12.  
  13. istream::istream(int size, char* p) : stream(new streambuf(p, size, size), TRUE)    // bind to string
  14. {
  15.     tied_to= 0;
  16. }
  17.  
  18. istream::istream(int fd, ostream* t) : stream(new filebuf(fd), TRUE)        // bind to file
  19. {
  20.     tied_to= t;
  21. }
  22.  
  23. istream::istream(char *path) : stream(new filebuf(path, input), TRUE)        // bind to file
  24. {
  25.     tied_to= 0;
  26.     if (! ((filebuf*)bp)->isopen())
  27.     state |= _fail;
  28. }
  29.  
  30. istream& istream::seek(long pos)
  31. {
  32.     bp->seek(pos, FALSE);
  33.     return *this;
  34. }
  35.  
  36. long istream::tell()
  37. {
  38.     return bp->tell(FALSE);
  39. }
  40.  
  41. ostream* istream::tie(ostream* s)
  42. {
  43.     ostream* t= tied_to;
  44.     tied_to= s;
  45.     return t; 
  46. }
  47.  
  48. void istream::eatwhite()
  49. {
  50.     flush();
  51.     register streambuf *nbp= bp;
  52.     register char c= nbp->sgetc();
  53.  
  54.     while (Isspace(zapeof(c)))
  55.     c= nbp->snextc();
  56.     if (c == EOF)
  57.     state |= _eof;
  58. }
  59.  
  60. int istream::read(u_char* s, int n)
  61. {
  62.     if (state || n <= 0) {
  63.     state |= _fail;
  64.     return 0;
  65.     }
  66.     int r= bp->sgetn((char*)s, n);
  67.     if (r < n)
  68.     setstate((state_value)(_fail|_eof));
  69.     return r;
  70. }
  71.  
  72. istream& istream::get(char& c)  // single character
  73. {
  74.     flush();
  75.     
  76.     if (state) {
  77.     state |= _fail;
  78.     return *this;
  79.     }
  80.     register tc= bp->sgetc();
  81.     if (tc == EOF)
  82.     state |= _fail|_eof;
  83.     else {
  84.     c= tc;
  85.     bp->stossc();
  86.     }
  87.     return *this;
  88. }
  89.  
  90. istream& istream::operator>> (register char& s)
  91. {
  92.     eatwhite();
  93.  
  94.     if (state) {
  95.     state |= _fail;
  96.     return *this;
  97.     }
  98.  
  99.     register c= bp->sgetc();
  100.     if (c == EOF)
  101.     state |= _fail|_eof;
  102.     else {
  103.     s= c;
  104.     bp->stossc();
  105.     }
  106.  
  107.     return *this;
  108. }
  109.  
  110. istream& istream::operator>> (register char* s)
  111. {
  112.     register streambuf *nbp = bp;
  113.  
  114.     eatwhite();
  115.  
  116.     if (state) {
  117.     state |= _fail;
  118.     return *this;
  119.     }
  120.  
  121.     /* get string */
  122.     register c= nbp->sgetc();
  123.     if (c == EOF)
  124.     state|= _fail;
  125.     while (!Isspace(c) && c != EOF) {
  126.     *s++= c;
  127.     c= nbp->snextc();
  128.     }
  129.     *s= '\0';
  130.  
  131.     if (c == EOF)
  132.     state|= _eof;
  133.  
  134.     return *this;
  135. }
  136.  
  137. istream& istream::operator>> (long &i)
  138. {
  139.     register int c, ii= 0;
  140.     register streambuf *nbp= bp;
  141.     int neg= 0;
  142.  
  143.     eatwhite();
  144.  
  145.     if (state) {
  146.     state |= _fail;
  147.     return *this;
  148.     }
  149.  
  150.     switch (c= nbp->sgetc()) {
  151.     case '-':
  152.     case '+':
  153.     neg= c;
  154.     c= nbp->snextc();
  155.     break;
  156.     case EOF:
  157.     state |= _fail;
  158.     }
  159.  
  160.     if (Isdigit(c)) {
  161.     do {
  162.         ii= ii*10+c-'0';
  163.     } while (Isdigit(c=nbp->snextc()));
  164.     i= (neg == '-') ? -ii : ii;
  165.     } else
  166.     state |= _fail;
  167.  
  168.     if (c == EOF)
  169.     state |= _eof;
  170.     return *this;
  171. }
  172.  
  173. istream& istream::operator>> (int &i)
  174. {
  175.     long int l;
  176.  
  177.     eatwhite();
  178.  
  179.     if (state) {
  180.     state |= _fail;
  181.     return *this;
  182.     }
  183.  
  184.     if (*this >> l)
  185.     i= (int) l;
  186.     return *this;
  187. }
  188.  
  189. istream& istream::operator>> (short &i)
  190. {
  191.     long l;
  192.  
  193.     eatwhite();
  194.  
  195.     if (state) {
  196.     state |= _fail;
  197.     return *this;
  198.     }
  199.  
  200.     if (*this >> l)
  201.     i= (short) l;
  202.     return *this;
  203. }
  204.  
  205. istream& istream::operator>>(double &d)
  206. {
  207.     register int c= 0;
  208.     char buf[256];
  209.     register char* p= buf;
  210.     register streambuf* nbp= bp;
  211.     
  212.     eatwhite();
  213.  
  214.     if (state) {
  215.     state |= _fail;
  216.     return *this;
  217.     }
  218.  
  219.     /* get the sign */
  220.     switch (c= nbp->sgetc()) {
  221.     case EOF:
  222.     state= _eof|_fail;
  223.     return *this;
  224.     case '-':
  225.     case '+':
  226.     *p++= c;
  227.     c= bp->snextc();
  228.     }
  229.  
  230.     /* get integral part */
  231.     while (Isdigit(c)) {
  232.     *p++= c;
  233.     c= bp->snextc();
  234.     }
  235.  
  236.     /* get fraction */
  237.     if (c == '.') {
  238.         do {
  239.             *p++ = c;
  240.             c = bp->snextc();
  241.         } while (Isdigit(c));
  242.     }
  243.  
  244.     /* get exponent */
  245.     if (c == 'e' || c == 'E') {
  246.     *p++= c;
  247.     switch (c= nbp->snextc()) {
  248.     case EOF:
  249.         state= _eof|_fail;
  250.         return *this;
  251.     case '-':
  252.     case '+':
  253.         *p++ = c;
  254.         c = bp->snextc();
  255.     }
  256.     while (Isdigit(c)) {
  257.         *p++= c;
  258.         c= bp->snextc();
  259.     }
  260.     }
  261.  
  262.     *p= 0;
  263.     d= atof(buf);
  264.  
  265.     if (c == EOF)
  266.     state |= _eof;
  267.     return *this;
  268. }
  269.  
  270. istream& istream::operator>> (float &f)
  271. {
  272.     double d;
  273.  
  274.     eatwhite();
  275.  
  276.     if (state) {
  277.     state |= _fail;
  278.     return *this;
  279.     }
  280.  
  281.     if (*this >> d)
  282.     f= d;
  283.     return *this;
  284. }
  285.  
  286. istream& istream::get(
  287.     register char* s,       // character array to read into
  288.     register int len,       // size of character array
  289.     register char term      // character that terminates input
  290. ) {
  291.     register int c;
  292.     register streambuf *nbp= bp;
  293.  
  294.     eatwhite();
  295.  
  296.     if (state) {
  297.     state |= _fail;
  298.     return *this;
  299.     }
  300.  
  301.     if ((c= bp->sgetc()) == EOF) {
  302.     state |= _fail | _eof;
  303.     return *this;
  304.     }
  305.  
  306.     while (c != term && c != EOF && len > 1) {
  307.     *s++= c;
  308.     c= nbp->snextc();
  309.     len--;
  310.     }
  311.     *s= '\0';
  312.     if (c == EOF)
  313.     state |= _eof;
  314.     return *this;
  315. }
  316.  
  317. istream& istream::putback(register char c)
  318. {
  319.     bp->sputbackc(c);
  320.     return *this;
  321. }
  322.  
  323. istream& istream::get(
  324.     register streambuf &s,  // streambuf to input to
  325.     register char term      // termination character
  326. ){
  327.     register c;
  328.     register streambuf *nbp = bp;
  329.  
  330.     eatwhite();
  331.  
  332.     if (state) {
  333.     state |= _fail;
  334.     return *this;
  335.     }
  336.  
  337.     if ((c = bp->sgetc()) == EOF) {
  338.     state|= _fail | _eof;
  339.     return *this;
  340.     }
  341.  
  342.     while (c != term && c != EOF) {
  343.     if (s.sputc(c) == EOF)
  344.         break;
  345.     c= nbp->snextc();
  346.     }
  347.     if (c == EOF)
  348.     state|= _eof;
  349.     return *this;
  350. }
  351.  
  352. istream& istream::operator>> (register streambuf &s)
  353. {
  354.     register int c;
  355.     register streambuf *nbp= bp;
  356.     
  357.     eatwhite();
  358.  
  359.     if (state) {
  360.     state|= _fail;
  361.     return *this;
  362.     }
  363.  
  364.     if ((c= bp->sgetc()) == EOF) {
  365.     state|= _fail | _eof;
  366.     return *this;
  367.     }
  368.  
  369.      while (c != EOF) {
  370.     if (s.sputc(c) == EOF)
  371.         break;
  372.     c= nbp->snextc();
  373.     }
  374.     if (c == EOF)
  375.     state |= _eof;
  376.     return *this;
  377. }
  378.  
  379. istream& istream::get(u_char &b)
  380. {
  381.     flush();
  382.     
  383.     if (state) {
  384.     state |= _fail;
  385.     return *this;
  386.     }
  387.     register tc= bp->sgetc();
  388.     if (tc == EOF)
  389.     state |= _fail|_eof;
  390.     else {
  391.     b= (u_char) tc;
  392.     bp->stossc();
  393.     }
  394.     return *this;
  395. }
  396.  
  397. istream& istream::operator>> (u_char &s)
  398. {
  399.     eatwhite();
  400.  
  401.     if (state) {
  402.     state |= _fail;
  403.     return *this;
  404.     }
  405.  
  406.     register c= bp->sgetc();
  407.     if (c == EOF)
  408.     state |= _fail|_eof;
  409.     else {
  410.     s= (u_char) c;
  411.     bp->stossc();
  412.     }
  413.  
  414.     return *this;
  415. }
  416.